Fix screen not turning off while charging